#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# File: data_download.py
# Date: 2024/7/4
# Author: 抖音、快手、视频号：东哥策略交易，微信：dongtrader
# Description: 东哥量化，带你走上量化之路。


from stock_trader_flow import *
# pandas 打印显示设置
pd.set_option('display.width', None)  # 自动调整列宽
pd.set_option('display.max_columns', None)  # 显示所有列
pd.set_option('display.max_rows', None)  # 显示所有行

if __name__ == "__main__":
    print("欢迎使用东哥量化工具箱\n关注各平台@东哥量化\n东哥量化，带你走上量化之路。\n")
    start_time = datetime.datetime.now()
    loggingseting(f'程序开始运行时间：{start_time}')
    while True:
        print("读取交易策略")
        print(traderStrategy())
        try:
            num_input = int(input("\n请选择(数字)：\n1. 自动交易\n2. 早盘挂单\n3. 尾盘交易\n4. 收盘交易核对\n5. 退出\n"))
        except:
            print("输入错误，请重新输入")
            continue
        if num_input == 1:
            auto_trader()
        elif num_input == 2:
            early_sell()
        elif num_input == 3:
            last_trade()
        elif num_input == 4:
            trader_check()
        elif num_input == 5:
            break
        else:
            print("输入错误，请重新输入")

